package cn.pigeon.health.utils;


import com.google.gson.Gson;
import com.qiniu.common.QiniuException;
import com.qiniu.http.Response;
import com.qiniu.storage.BucketManager;
import com.qiniu.storage.Configuration;
import com.qiniu.storage.Region;
import com.qiniu.storage.UploadManager;
import com.qiniu.storage.model.DefaultPutRet;
import com.qiniu.util.Auth;

import java.io.UnsupportedEncodingException;

/**
 * 七牛云工具类
 */
public class QiniuTools {
    private static final String accessKey="_bS56CYiwkf_cO_1XiIcQ_nftdqS7o1OH2aHs1-Q";
    private static final String secretKey="VFCwzZvjdBlVkClZkNfVtGU3RMeTao1WIVx652kF";
    //存储空间的名称
    private static final String bucket="pigeon2019-space";


    /**
     * 按指定的本地路径上传文件
     * @param localFilePath
     */
    public static void uploadFile(String localFilePath,String fileName){
        uploadCommon(localFilePath,null,fileName);
    }

    /**
     * 按字节数组上传文件
     * @param uploadBytes
     * @param fileName
     */
    public static void uploadFile(byte[] uploadBytes,String fileName){
        uploadCommon(null,uploadBytes,fileName);
    }

    /**
     * 通用上传文件方法
     * @param localFilePath
     * @param uploadByts
     * @param fileName
     */
    private static void uploadCommon(String localFilePath,byte[] uploadByts,String fileName){
        //构造一个带指定 Region 对象的配置类
        Configuration cfg = new Configuration(Region.region2());
        //...其他参数参考类注释
        UploadManager uploadManager = new UploadManager(cfg);


        Auth auth = Auth.create(accessKey, secretKey);
        String upToken = auth.uploadToken(bucket);
        Response response = null;
        try {
            //判断是按文件路径传递，还是按字节数组传递
            if(localFilePath==null || localFilePath.length()<=0){
                response = uploadManager.put(uploadByts, fileName, upToken);
            }else {
                response = uploadManager.put(localFilePath, fileName, upToken);
            }
            //解析上传成功的结果
            DefaultPutRet putRet = new Gson().fromJson(response.bodyString(), DefaultPutRet.class);
            System.out.println(putRet.key);
            System.out.println(putRet.hash);
        } catch (QiniuException ex) {
            Response r = ex.response;
            System.err.println(r.toString());
            try {
                System.err.println(r.bodyString());
            } catch (QiniuException ex2) {
                //ignore
            }
        }
    }


    /**
     * 删除文件
     * @param fileName
     */
    public static void deleteFile(String fileName){
        //构造一个带指定 Region 对象的配置类
        Configuration cfg = new Configuration(Region.region2());
        Auth auth = Auth.create(accessKey, secretKey);
        BucketManager bucketManager = new BucketManager(auth, cfg);
        try {
            bucketManager.delete(bucket, fileName);
        } catch (QiniuException ex) {
            //如果遇到异常，说明删除失败
            System.err.println(ex.code());
            System.err.println(ex.response.toString());
        }
    }
}
